from langchain.chains.router import MultiPromptChain


# def test():
#     # 构建路由chain
#     _3_chain = MultiPromptChain(
#         router_chain=router_chain,
#         destination_chains=chain_map,
#         default_chain=default_chain,
#         verbose=True
#     )
#
#     # 测试路由功能
#     print(_3_chain.run("如何为玫瑰浇水?"))  # 路由到养护chain
#     print(_3_chain.run("如何布置婚礼花艺?"))  # 路由到装饰chain
#     print(_3_chain.run("今天天气如何?"))  # 使用默认chain

